import random

# 定义一个列表,表示袋中的球
balls = ['W', 'W', 'W', 'W', 'R', 'R']


# 定义一个函数,用于模拟放回抽样
def simulate_with_replacement(balls, num_samples):
    # 定义变量,用于记录白球和红球的数量
    white_count = 0
    red_count = 0
    both_white_count = 0
    both_red_count = 0

    # 循环模拟放回抽样
    for i in range(num_samples):
        # 从袋子中随机抽取一个球
        ball1 = random.choice(balls)
        # 如果抽取到白球,则白球数量加1
        if ball1 == 'W':
            white_count += 1
        # 如果抽取到红球,则红球数量加1
        else:
            red_count += 1

        # 从袋子中随机抽取一个球
        ball2 = random.choice(balls)
        # 如果两个球相同,则判断是否是白球
        if ball1 == ball2:
            # 如果两个球都是白球,则白球数量加1
            if ball1 == 'W':
                both_white_count += 1
            # 如果两个球都是红球,则红球数量加1
            elif ball1 == 'R':
                both_red_count += 1

    # 返回白球数量,红球数量,白球和红球数量同时出现的数量
    return white_count, red_count, both_white_count, both_red_count


# 调用函数,模拟取球过程
white_count, red_count, both_white_count, both_red_count = simulate_with_replacement(balls=balls, num_samples=1000)

# 输出结果
print("取到两个都是白球的概率为:", both_white_count / 1000)
print("取到两个都是红球的概率为:", both_red_count / 1000)